import time
import unittest
from selenium import webdriver
from Unit_01.Page import Page
class TestCase(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
    def tearDown(self):
        time.sleep(10)
        self.driver.quit()
    def test_baidu_01(self):
        po = Page(self.driver)
        po.baidu_luoji()
if __name__ == '__main__':
    unittest.main()